import os

def get_size(path):
    """
    返回给定路径的大小（如果是文件，则返回文件大小；如果是目录，则返回目录下所有文件的总大小）。
    :param path: 文件或目录的路径
    :return: 文件或目录的大小（字节）
    """
    if os.path.isfile(path):
        return os.path.getsize(path)  # 获取文件大小
    else:
        return None
    
# 使用函数
file_or_dir_path = r"D:\photo\theme\p2024\IMG_3672.JPEG"  # 替换为你的文件或目录路径
size = get_size(file_or_dir_path)
print("size: ",size)